// Copyright (c) 2021-2024 SigScalr, Inc.
//
// This file is part of SigLens Observability Solution
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

package ingestserver

import (
	"github.com/siglens/siglens/pkg/hooks"
	log "github.com/sirupsen/logrus"
	"github.com/valyala/fasthttp"
)

func (hs *ingestionServerCfg) Recovery(next func(ctx *fasthttp.RequestCtx)) func(ctx *fasthttp.RequestCtx) {
	fn := func(ctx *fasthttp.RequestCtx) {
		if hook := hooks.GlobalHooks.IngestMiddlewareRecoveryHook; hook != nil {
			err := hook(ctx)
			if err != nil {
				log.Errorf("ingestionServerCfg.Recovery: got error from hook: %v", err)
				return
			}
		}

		// do next
		next(ctx)
	}
	return fn
}
